#ifndef __COMM_HPP
#define __COMM_HPP

#include <iostream>
#include <sys/ipc.h>
#include <sys/shm.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <cerrno>
#include <cstdio>
#include <cstring>
#include <unistd.h>
#include <cassert>

using namespace std;

#define PATHNAME "."
#define PROJID 0x6666

const int gsize = 4096;//共享内存的大小是以PAGE页（4KB）为单位的
//用户需求多少操作系统就分配多少，最小单位是4KB，用户即使无需这些空间，操作系统也至少4KB大小


key_t GetKey()
{
    key_t k = ftok(PATHNAME, PROJID);
    if (k == -1)
    {
        cerr << errno << ":" << strerror(errno) << endl;
        exit(1);
    }
    return k;
}

static int createShmHelper(key_t k, int size, int flag)
{
    int shmid = shmget(k, size, flag);
    if (shmid == -1)
    {
        cerr << "error: " << errno << " : " << strerror(errno) << endl;
        exit(2);
    }
    return shmid;
}

int createShm(key_t k, int size)//创建共享内存
{
    umask(0);
    return createShmHelper(k, size, IPC_CREAT | IPC_EXCL|0666);//增加0666，目的是让创造该共享内存者具有权限，否则使用IPC_STAT受阻
}

int getShm(key_t k, int size)//获取共享内存
{
    return createShmHelper(k, size, IPC_CREAT);
}

void deleteshm(int shmid)//删除共享内存
{
    int n = shmctl(shmid, IPC_RMID, nullptr);
    assert(n != -1);
    (void)n;
}

char* attachshm(int shmid)//关联共享内存
{
    char* start=(char*)shmat(shmid,NULL,0);\
    //shamat与malloc有些相似，不过malloc是分配一块内存空间使用
    //而shmat是获取一块内存空间起始地址，这个空间在函数执行之前已经存在
    return start;
}

void detachshm(char* start)
{
    int n=shmdt(start);
    assert(n!=-1);
    (void)n;
}


#define SERVER 1
#define CLIENT 0

//可以将共享内存封装为一个类，实现自动关联、断联与释放
class Init
{
public:
    Init(int t)
    :type(t)
    {
        key_t k = GetKey();
        if(type == SERVER) shmid = createShm(k, gsize);
        else shmid = getShm(k, gsize);
        start = attachshm(shmid);
    }
    char *getStart(){ return start; }
    ~Init()
    {
        detachshm(start);
        if(type == SERVER) deleteshm(shmid);
    }
private:
    char* start;
    int type;//表示服务端和用户端
    int shmid;
};

#endif